package com.ebike.servlet;

import java.io.File;
import java.io.IOException;
import java.io.PrintWriter;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.List;
import java.util.UUID;

import javax.servlet.ServletException;
import javax.servlet.annotation.WebServlet;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import org.apache.commons.fileupload.FileItem;
import org.apache.commons.fileupload.disk.DiskFileItemFactory;
import org.apache.commons.fileupload.servlet.ServletFileUpload;

import com.ebike.comm.util.PictureUtil;

@WebServlet("/Upload")
public class Upload extends HttpServlet {

	private static final long serialVersionUID = 1L;

	// 上传文件存储目录
	private static final String UPLOAD_DIRECTORY = "upload";
	
	//	private static final String HEAD = "http://106.15.42.5:8081/upload/";
	
	private static final String HEAD = "http://47.97.221.244:8081/upload/";
	

	// 上传配置
	private static final int MEMORY_THRESHOLD = 1024 * 1024 * 3; // 3MB
	private static final int MAX_FILE_SIZE = 1024 * 1024 * 40; // 40MB
	private static final int MAX_REQUEST_SIZE = 1024 * 1024 * 50; // 50MB

	public void doGet(HttpServletRequest request, HttpServletResponse response)
			throws ServletException, IOException {

	}

	public void doPost(HttpServletRequest request, HttpServletResponse response)
			throws ServletException, IOException {

		String ori_url = "";
		String big_url = "";
		String small_url = "";
		// 检测是否为多媒体上传
		if (!ServletFileUpload.isMultipartContent(request)) {
			// 如果不是则停止
			PrintWriter out = response.getWriter();
			out.write("Error: 表单必须包含 enctype=multipart/form-data");
			return;
		}
		// 配置上传参数
		DiskFileItemFactory factory = new DiskFileItemFactory();
		// 设置内存临界值 - 超过后将产生临时文件并存储于临时目录中
		factory.setSizeThreshold(MEMORY_THRESHOLD);
		// 设置临时存储目录
		factory.setRepository(new File(System.getProperty("java.io.tmpdir")));
		ServletFileUpload upload = new ServletFileUpload(factory);
		// 设置最大文件上传值
		upload.setFileSizeMax(MAX_FILE_SIZE);
		// 设置最大请求值 (包含文件和表单数据)
		upload.setSizeMax(MAX_REQUEST_SIZE);
		// 中文处理
		upload.setHeaderEncoding("UTF-8");
		// 构造临时路径来存储上传的文件
		// 这个路径相对当前应用的目录
		SimpleDateFormat df = new SimpleDateFormat("yyyyMMdd");//设置日期格式
		String dateS = df.format(new Date());
		String uploadPath = request.getServletContext().getRealPath("./../")
				+ File.separator + UPLOAD_DIRECTORY + File.separator + dateS;

		// 如果目录不存在则创建
		File uploadDir = new File(uploadPath);
		if (!uploadDir.exists()) {
			uploadDir.mkdir();
		}
		try {
			
		
			// 解析请求的内容提取文件数据
			@SuppressWarnings("unchecked")
			List<FileItem> formItems = upload.parseRequest(request);
			if (formItems != null && formItems.size() > 0) {
				// 迭代表单数据
				for (FileItem item : formItems) {
					// 处理不在表单中的字段
					if (!item.isFormField()) {
						String uuid = UUID.randomUUID().toString();
						String fileName = uuid + "_ori.jpg"; // 以uuid命名
						String filePath = uploadPath + File.separator
								+ fileName;
						File storeFile = new File(filePath);
						// 在控制台输出文件的上传路径
						System.out.println(filePath);
						// 保存文件到硬盘
						item.write(storeFile);
						ori_url = HEAD + dateS + "/" + fileName;
						
						int[] wh = PictureUtil.getImgWidth(storeFile);
						// 生成大图
						if (wh[0] > 720) {
							float rate = (float)720 / wh[0];
							System.out.println("开始生成大图，压缩比例：" + rate);
							String bfileName = uuid + "_big.jpg";
							String distPath = uploadPath + File.separator + bfileName;
							PictureUtil.reduceImg(storeFile, distPath, 0,
									0, rate);
							big_url = HEAD + dateS + "/" + bfileName;
							System.out.println("开始生成大图完成");
						} else {
							// 宽度小于720 原图就是大图
							big_url = ori_url;
						}

						// 生成小图
						if (wh[0] > 360) {
							float rate = (float)360 / wh[0];
							System.out.println("开始生成小图，压缩比例：" + rate);
							String sfileName = uuid + "_small.jpg";
							String distPath = uploadPath + File.separator + sfileName;
							PictureUtil.reduceImg(storeFile, distPath, 0,
									0, rate);
							small_url = HEAD + dateS + "/" + sfileName;
							System.out.println("开始生成小图完成");
						} else {
							// 宽度小于360 原图就是小图
							small_url = ori_url;
						}

					}
				}
			}
		} catch (Exception ex) {
			request.setAttribute("message", "错误信息: " + ex.getMessage());
		}

		PrintWriter out = response.getWriter();
		out.write("{\"ori_url\":\"" + ori_url + "\",\"big_url\":\"" + big_url
				+ "\",\"small_url\":\"" + small_url + "\"}");

	}
}
